data1 = torch.FloatTensor([[0, 1, 2, 3, 4],
[5, 6, 7, 8, 9],
[10, 11, 12, 13, 14]])
print(data1, data1.shape)
print(data1[1,2])
print(data1[:, 1])
print(data1[:,-1])
print(data1[:, :]) #전체tensor([[ 0., 1., 2., 3., 4.],
[ 5., 6., 7., 8., 9.],
[10., 11., 12., 13., 14.]]) torch.Size([3, 5])
tensor(7.)
tensor([ 1., 6., 11.])
tensor([ 4., 9., 14.])
tensor([[ 0., 1., 2., 3., 4.],
[ 5., 6., 7., 8., 9.],
[10., 11., 12., 13., 14.]])